import subprocess as sp
import os
import sys

""" Archivo que obtiene las resoluciones de una m3u y genera un csv.
    Recibe:
        1. Ruta y nombre del archivo M3U
    Genera:
        1. Un archivo csv con el mismo nombre y en la misma ruta
"""

def obtainResolutions(url):
    data = str(sp.check_output(['youtube-dl', '-F', url]))
    dataList = data.split('\\n')
    res = [line.split(' ')[0] for line in dataList
        if 'hls' in line]
    return res

if __name__ == '__main__':
    f = open(sys.argv[1],'r')
    serie_resolutions = dict()
    chapter_no = 0
    for line in f.readlines():
        if "#EXTM3U" in line:
            pass
        elif '#EXTINF' not in line:
            res = obtainResolutions(line)
            serie_resolutions['Capitulo '+str(chapter_no)] = res
        else:
            chapter_no = chapter_no+1
    f.close()
    print(serie_resolutions)
    f = open(sys.argv[1].replace('m3u', 'csv'),'w')
    for key in serie_resolutions:
        line = key
        for res in serie_resolutions[key]:
            line = line+';'+res
        f.write(line+'\n')
    f.close()